package db

import (
	"dc-news/common/global/setting"
	"fmt"
	"go.uber.org/zap"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"moul.io/zapgorm2"
	"time"
)

var DB *gorm.DB

func InitMySQL() (err error) {
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		setting.Conf.MySQLConfig.User,
		setting.Conf.MySQLConfig.Password,
		setting.Conf.MySQLConfig.Host,
		setting.Conf.MySQLConfig.Port,
		setting.Conf.MySQLConfig.DB,
	)
	logger := zapgorm2.New(zap.L())
	logger.SetAsDefault()
	DB, err = gorm.Open(mysql.Open(dsn))
	if err != nil {
		return
	}
	// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
	sqlDB, err := DB.DB()
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(10)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(100)
	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)
	return nil
}
